import argparse

args = argparse.ArgumentParser()
args.add_argument('--data_name', type=str, default='Freebase')
args.add_argument('--seed', type=int, default=0)
args = args.parse_args()

from utilities.train import train_
import yaml
with open('./cfg/Freebase.yaml', 'r') as f:
    config = yaml.load(f, Loader=yaml.FullLoader)

best_pre, label = train_(config, args=args)
from Clustering_metrics import cal_metrics
print(cal_metrics(best_pre, label))
